#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>
typedef unsigned char u8;
typedef unsigned int  u32;
void read_bin(char *path, u8 *buf, u8 size)
{
    FILE *infile;

    if((infile = fopen(path, "rb")) == NULL)
    {
        printf( "\nCan not open the path: %s \n", path);
        exit(-1);
    }

    fread(buf, sizeof(u8), size, infile);
    fclose(infile);
}
u32 GetBinSize(char *filename)
{
    u32  siz = 0;
    FILE  *fp = fopen(filename, "rb");

    if (fp)
    {
        fseek(fp, 0, SEEK_END);
        siz = ftell(fp);
        fclose(fp);
    }

    return siz;
}
void OutPut_handle(char *outpath, u8 *buf, u32 size)
{
    FILE *infile;
    int i, j, k, n;
    int fd ;
    char pbuf[10] = {0};
    char mfgimage[4096 * 2];
    char *array = "static const unsigned char SPIflashimage[SPIIMAGESIZE] = {\n";
    char *array1 = "static const unsigned char mfgimage[MFGIMAGESIZE] = {\n";
    char *Handle = "#ifndef SPI_FLASH_H_ \n";
    char *Handle1 = "#define SPI_FLASH_H_ \n";
    char *SPI_SPIflash = "#define SPI_SPIflash 0 \n";
    char *SPIIMAGESIZE = "#define SPIIMAGESIZE   411652 \n";
    char *MFGIMAGESIZE = "#define MFGIMAGESIZE   411652 \n";
    char *SIZE_4K      = "#define SIZE_4K   4096*2 \n";
    char *line_T       = "\n";
    char *EndIF        = "\n#endif \n";

    if((infile = fopen(outpath, "wa+")) == NULL)
    {
        printf( "\nCan not open the path: %s \n", outpath);
        exit(-1);
    }

    k = 0;
    fwrite(Handle, strlen(Handle), 1, infile);
    fwrite(Handle1, strlen(Handle1), 1, infile);
    fwrite(SPI_SPIflash, strlen(SPI_SPIflash), 1, infile);
    fwrite(SPIIMAGESIZE, strlen(SPIIMAGESIZE), 1, infile);
    fwrite(MFGIMAGESIZE, strlen(MFGIMAGESIZE), 1, infile);
    fwrite(SIZE_4K, strlen(SIZE_4K), 1, infile);
    fwrite(array, strlen(array), 1, infile);

    for(i = 0; i < size; i++)
    {
        k++;
        sprintf(pbuf, "0x%02x", buf[i]);
        fwrite(pbuf, strlen(pbuf), 1, infile);

        if(k != 16)
            fwrite(", ", strlen(", "), 1, infile);
        else
            fwrite(",", strlen(","), 1, infile);

        if(k == 16)
        {
            k = 0;
            fwrite("\n", strlen("\n"), 1, infile);
        }
    }

    fseek(infile, 0, SEEK_END);

    if(k == 0)
        fwrite("};", strlen("};"), 1, infile);
    else
        fwrite("\n};", strlen("\n};"), 1, infile);

    //在infile文件中和换行
    fwrite(line_T, strlen(line_T), 1, infile);
    //创建一个文件用于保存零数组
    fd = creat("nufile.bin", 0777);

    if(-1 == fd)
    {
        perror("creat fair!");
        return ;
    }

    //偏移写空
    int offset = lseek(fd, 4096 * 2, SEEK_END);
    write(fd, "", 1);

    /**************************************************/
    //清数组
    for(i = 0 ; i < 10 ; i++)
        pbuf[i] = 0 ;

    for(i = 0 ; i < 4096 * 2 ; i++)
        mfgimage[i] = 0 ;

    //写第二个数组
    fwrite(array1, strlen(array1), 1, infile);
    //从空文件里读数据读到mfgimage数组
    read(fd, mfgimage, 4096 * 2);
    //关闭文件句柄
    close(fd);
    //往文件后面继续写数据
    k = 0 ;

    for(i = 0; i < 4096 * 2; i++)
    {
        k++;
        sprintf(pbuf, "0x%02x", mfgimage[i]);
        fwrite(pbuf, strlen(pbuf), 1, infile);

        if(k != 16)
            fwrite(", ", strlen(", "), 1, infile);
        else
            fwrite(",", strlen(","), 1, infile);

        if(k == 16)
        {
            k = 0;
            fwrite("\n", strlen("\n"), 1, infile);
        }
    }

    fseek(infile, 0, SEEK_END);

    if(k == 0)
        fwrite("};", strlen("};"), 1, infile);
    else
        fwrite("\n};", strlen("\n};"), 1, infile);

    fwrite(line_T, strlen(line_T), 1, infile);
    fwrite(EndIF, strlen(EndIF), 1, infile);

    //删除当前目录下的一个空洞文件
    if(remove("nufile.bin") == 0)
        printf("del file success!\n");
    else
        printf("del file fair!\n");

    fclose(infile);
}

int main()
{
    u8 *buf = NULL;
    u32 size;
    char srcbin[100] = {0};
    char dstfile[100] = {0};
    //读取目标.bin文件
    printf("please input src file path\n");
    scanf("%s", srcbin);
    //创建一个.h头文件用于保存bin转C数组的文件
    printf("please input output path\n");
    scanf("%s", dstfile);
    //获取文件的大小
    size = GetBinSize(srcbin);
    //申请用于存放该文件的数组
    buf = (unsigned char *)malloc(sizeof(unsigned char) * size);
    //读取文件
    read_bin(srcbin, buf, size);
    //制作头文件，该头文件下含有两个数组，一个是有数据的，另外一个是全0数组
    //全0主要备用，以后要清空可以调用这个数组
    OutPut_handle(dstfile, buf, size);
    return 0;
}